//////////////////////////////////////////////////////////////////////////////
//////////////////////////////////////////////////////////////////////////////
//
// copyright            : (C) 2013 by Eran Ifrah
// file name            : bitmap_loader.h
//
// -------------------------------------------------------------------------
// A
//              _____           _      _     _ _
//             /  __ \         | |    | |   (_) |
//             | /  \/ ___   __| | ___| |    _| |_ ___
//             | |    / _ \ / _  |/ _ \ |   | | __/ _ )
//             | \__/\ (_) | (_| |  __/ |___| | ||  __/
//              \____/\___/ \__,_|\___\_____/_|\__\___|
//
//                                                  F i l e
//
//    This program is free software; you can redistribute it and/or modify
//    it under the terms of the GNU General Public License as published by
//    the Free Software Foundation; either version 2 of the License, or
//    (at your option) any later version.
//
//////////////////////////////////////////////////////////////////////////////
//////////////////////////////////////////////////////////////////////////////

#ifndef BITMAP_LOADER_H
#define BITMAP_LOADER_H

#include "cl_command_event.h"
#include "codelite_exports.h"
#include "fileextmanager.h"
#include "wxStringHash.h"

#include <vector>
#include <wx/bitmap.h>
#include <wx/filename.h>
#include <wx/imaglist.h>

class WXDLLIMPEXP_SDK clMimeBitmaps
{
    /// Maps between image-id : index in the list
    std::unordered_map<int, int> m_fileIndexMap;
    std::vector<wxBitmap> m_bitmaps;
    std::vector<wxBitmap> m_disabled_bitmaps;

public:
    clMimeBitmaps() = default;
    ~clMimeBitmaps() = default;

    /**
     * @brief return the bitmap index that matches a given file type
     */
    int GetIndex(int type, bool disabled = false) const;
    /**
     * @brief return the bitmap index that matches the given filename
     */
    int GetIndex(const wxString& filename, bool disabled = false) const;
    const wxBitmap& GetBitmap(int type, bool disabled) const;
    void AddBitmap(const wxBitmap& bitmap, int type);
    void Clear();
    bool IsEmpty() const { return m_bitmaps.empty(); }
    std::vector<wxBitmap>& GetBitmaps() { return m_bitmaps; }
    const std::vector<wxBitmap>& GetBitmaps() const { return m_bitmaps; }

    // call this in order to merge the disabled and active bitmap lists
    void Finalise();
};

class WXDLLIMPEXP_SDK clBitmaps;

class WXDLLIMPEXP_SDK clBitmapList : public wxEvtHandler
{
    size_t m_index = 0;
    struct BmpInfo {
        wxBitmap* bmp_ptr = nullptr;          // if this is set, it means that it is part of the BitmapLoader class
        wxBitmap bmp = wxNullBitmap;          // user provided bitmap
        wxBitmap bmp_disabled = wxNullBitmap; // this one is always a one that we provide
        wxString name;
        int ref_count = 1;
    };
    std::unordered_map<size_t, BmpInfo> m_bitmaps;
    std::unordered_map<wxString, size_t> m_nameToIndex;

protected:
    void OnBitmapsUpdated(clCommandEvent& event);
    size_t FindIdByName(const wxString& name) const;
    size_t DoAdd(const wxBitmap& bmp, const wxBitmap& bmpDisabled, const wxString& bmp_name, bool user_bmp);

public:
    clBitmapList();
    virtual ~clBitmapList();
    size_t Add(const wxBitmap& bmp, const wxString& name);
    size_t Add(const wxString& bmp_name, int size = wxNOT_FOUND);
    size_t size() const { return m_bitmaps.size(); }
    bool empty() const { return m_bitmaps.empty(); }
    const wxBitmap& Get(size_t index, bool disabledBmp);
    const wxBitmap& Get(const wxString& name, bool disabledBmp);
    void Delete(size_t index);
    void Delete(const wxString& name);
    void clear();
    /**
     * @brief return bitmap name for a given index
     */
    const wxString& GetBitmapName(size_t index) const;
};

class WXDLLIMPEXP_SDK BitmapLoader : public wxEvtHandler
{
public:
    friend class clBitmaps;
    using BitmapMap_t = std::unordered_map<FileExtManager::FileType, wxBitmap>;
    using Vec_t = std::vector<wxBitmap>;

    enum eBitmapId {
        kClass = 1000,
        kStruct,
        kNamespace,
        kTypedef,
        kMemberPrivate,
        kMemberProtected,
        kMemberPublic,
        kFunctionPrivate,
        kFunctionProtected,
        kFunctionPublic,
        kEnum,
        kCEnum,
        kEnumerator,
        kConstant,
        kMacro,
        kCxxKeyword,
        kClose,
        kSave,
        kSaveAll,
        kTable,
        kDatabase,
        kColumn,
        kFind,
        kAngleBrackets,
        kSort,
    };

    clMimeBitmaps& GetMimeBitmaps() { return m_mimeBitmaps; }
    const clMimeBitmaps& GetMimeBitmaps() const { return m_mimeBitmaps; }

    /**
     * @brief prepare an image list allocated on the heap which is based on
     * the FileExtManager content. It is the CALLER responsibility for deleting the memory
     */
    BitmapLoader::Vec_t* GetStandardMimeBitmapListPtr() { return &GetMimeBitmaps().GetBitmaps(); }

    /**
     * @brief return the image associated with a filename
     */
    const wxBitmap& GetBitmapForFile(const wxFileName& filename, bool disabled) const
    {
        return GetBitmapForFile(filename.GetFullName(), disabled);
    }
    const wxBitmap& GetBitmapForFile(const wxString& filename, bool disabled) const;

    /**
     * @brief return the image index in the image list prepared by GetStandardMimeBitmapListPtr()
     * @return wxNOT_FOUND if no match is found, the index otherwise
     */
    int GetMimeImageId(const wxString& filename, bool disabled = false);

    /**
     * @brief return bitmap bundle by name
     */
    const wxBitmapBundle& GetBundle(const wxString& name) const;

    /**
     * @brief return the image index in the image list prepared by GetStandardMimeBitmapListPtr()
     * @return wxNOT_FOUND if no match is found, the index otherwise
     */
    int GetMimeImageId(int type, bool disabled = false);
    int GetImageIndex(int type, bool disabled = false) { return GetMimeImageId(type, disabled); }
    const wxBitmap& LoadBitmap(const wxString& name, int requestedSize = 16);
    bool GetIconBundle(const wxString& name, wxIconBundle* bundle);
    /**
     * @brief Returns a wxImageList containing the standard MIME bitmaps.
     *
     * @details This member function creates a new wxImageList, adds all
     * bitmaps from @c GetStandardMimeBitmapListPtr() to it, and returns
     * the list. The caller takes ownership of the returned pointer and
     * is responsible for deleting it when it is no longer needed.
     *
     * @return A newly allocated wxImageList with the standard MIME bitmaps
     * (may be @c nullptr if allocation fails).
     *
     * @see GetStandardMimeBitmapListPtr()
     *
     * @code
     * // Example usage
     * wxImageList* mimeList = BitmapLoader::GetStandardMimeImageList();
     * // ... use mimeList ...
     * delete mimeList;   // caller must free the list
     * @endcode
     */
    wxImageList* GetStandardMimeImageList();

protected:
    wxIcon GetIcon(const wxBitmap& bmp) const;
    void CreateMimeList();

private:
    BitmapLoader(wxWindow* win, bool darkTheme);
    virtual ~BitmapLoader() = default;

    void AddBitmapInternal(const wxBitmapBundle& bundle, const wxString& base_name);
    void Initialize(bool darkTheme);
    void LoadSVGFiles(bool darkTheme);

    wxFileName m_zipPath;
    std::unordered_map<wxString, wxBitmap> m_toolbarsBitmaps;
    std::unordered_map<wxString, wxString> m_manifest;
    std::unordered_map<int, int> m_fileIndexMap;
    clMimeBitmaps m_mimeBitmaps;
    std::unordered_map<wxString, wxBitmapBundle>* GetBundles(bool darkTheme) const;
    wxWindow* m_win{nullptr};
};

wxDECLARE_EXPORTED_EVENT(WXDLLIMPEXP_SDK, wxEVT_BITMAPS_UPDATED, clCommandEvent);
class WXDLLIMPEXP_SDK clBitmaps : public wxEvtHandler
{
    BitmapLoader* m_lightBitmaps = nullptr;
    BitmapLoader* m_darkBitmaps = nullptr;
    BitmapLoader* m_activeBitmaps = nullptr;

protected:
    void InitialiseInternal(wxWindow* win);
    void OnSysColoursChanged(clCommandEvent& event);

protected:
    clBitmaps();
    virtual ~clBitmaps();

public:
    static clBitmaps& Get();
    static void Initialise(wxWindow* win);

    BitmapLoader* GetLoader();
    void SysColoursChanged();
};

/// Helper load function
WXDLLIMPEXP_SDK void
clLoadSidebarBitmap(const wxString& name, wxWindow* win, wxBitmap* light_theme_bmp, wxBitmap* dark_theme_bmp);

/// Clear the sidebar bitmaps cache
void WXDLLIMPEXP_SDK clClearSidebarBitmapCache();
#endif
